import sys
import os
import shutil
from datetime import datetime

def extract_embedded_files():
    # 获取打包后的资源目录（仅exe运行时有效）
    if hasattr(sys, '_MEIPASS'):
        resource_dir = sys._MEIPASS
    else:
        resource_dir = os.path.abspath('.') 

    # 定义要提取的文件（原路径→目标路径）
    files_to_extract = [
        (os.path.join(resource_dir, 'src/main_qwidget_ui.py'), './main_qwidget_ui.py'),
        (os.path.join(resource_dir, 'src/main_qwidget.ui'), './main_qwidget.ui'),
        (os.path.join(resource_dir, 'src/main.py'), './main.py'),
        #(os.path.join(resource_dir, 'doc', 'file2.pdf'), './doc/file2_copy.pdf') 可以指定目录
    ]


    for src_path, dst_path in files_to_extract:
        # 创建目标目录（如果不存在）
        dst_dir = os.path.dirname(dst_path)
        if not os.path.exists(dst_dir):
            os.makedirs(dst_dir)
        
        
        shutil.copy2(src_path, dst_path)
        current_time = datetime.now().strftime("%H:%M:%S")
        print(f"{current_time}    \033[32mINFO\033[0m    {src_path} → {dst_path}")

if __name__ == "__main__":
    extract_embedded_files()